Custom aggregation to a JSON in pyspark